# Copyright (c) 2025 Bytedance Ltd. and/or its affiliates
# SPDX-License-Identifier: MIT

from pydantic import BaseModel, Field

from src.graph.talent_model import Talent


class TalentResponse(BaseModel):
    """Response model for RAG config."""

    talents: list[Talent] = Field(..., description="人才库列表")


class TalentRequest(BaseModel):
    """Request model for RAG resource."""

    thread_id: str | None = Field(
        None, description="线程id"
    )
